#! /usr/bin/env python
# coding:utf-8
'''该模块用于建数据表！！'''

def creat_grade():
    from function import db, cursor
    select_YorN = raw_input('请注意！！若数据表已经存在，则该选项会还原数据表，是否要还原？（Y/N)：')
    if (select_YorN == 'Y') or (select_YorN == 'y'):
        stu_grade = [{'学号': 11011, '姓名': '徐灿灿', '语文' :50, '数学' : 69, '英语': 98},
                     {'学号': 11012, '姓名': '王培中', '语文' : 73, '数学' : 68, '英语':50},
                     {'学号': 11013, '姓名': '郭靖', '语文' : 69, '数学' : 54, '英语':72},
                     {'学号': 11014, '姓名': '黄青丝', '语文' : 80, '数学' : 69, '英语': 50},
                     {'学号': 11015, '姓名': '薛如霜', '语文' : 60, '数学' : 59, '英语': 70},
                     {'学号': 11016, '姓名': '邓李', '语文' : 40, '数学' : 49, '英语': 60},
                     {'学号': 11017, '姓名': '孟青山', '语文' : 70, '数学' : 79, '英语': 40}]

        cursor.execute('DROP TABLE IF EXISTS studata')
        sql =''' CREATE TABLE `studata` (
          `id` bigint(255) NOT NULL,
          `name` varchar(255) NOT NULL,
          `chinese` int(10) DEFAULT NULL,
          `math` int(10) DEFAULT NULL,
          `english` int(10) DEFAULT NULL
        ) ENGINE=InnoDB DEFAULT CHARSET=utf8;'''
        cursor.execute(sql)
        cursor.execute('set names utf8')
        for i in stu_grade:
            comm = "INSERT INTO studata(id, name, chinese, math, english )\
                            VALUES (%d, '%s', %d, %d, %d)" % (i['学号'], i['姓名'], i['语文'], i['数学'], i['英语'])
            cursor.execute(comm)
        db.commit()
        print '创建成功，请不要重复创建！！'
